package main

import (
	"os"

	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
	"sweeter.star/go-demo/common"
)

func main()  {
	gin.SetMode(gin.ReleaseMode)
	// 读取项目配置项
	InitConfig()
	// 初始化数据连接池
	common.InitDB()

	r := gin.Default()
	r = router(r)		// 导入路由

	// 自定义启动端口号
	port := viper.GetString("server.port")
	if port != "" {
		panic(r.Run(":" + port))
	}
	panic(r.Run())
}


func InitConfig()  {
	workdir, _ := os.Getwd()
	viper.SetConfigName("app")
	viper.SetConfigType("yml")
	viper.AddConfigPath(workdir + "/config")
	err := viper.ReadInConfig()

	if err != nil {
		panic(err)
	}
}